import os
import numpy as np
from scipy.special import gammaln 


def gen_golden_data_lgamma():
    dtype = np.float16
    input_shape = [8, 2048]
    output_shape = [8, 2048]

    x = np.random.uniform(1, 3, input_shape).astype(dtype)
    
    golden = gammaln(x).astype(dtype)

    os.system("mkdir -p input")
    os.system("mkdir -p output")
    x.tofile("./input/input_x.bin")
    golden.tofile("./output/golden.bin")


if __name__ == "__main__":
    gen_golden_data_lgamma()